const WebSocket = require('ws');

function startWebSocket(server) {
    const ws = new WebSocket.Server({ server });
    //处理客户端与服务器成功建立 WebSocket 连接时触发的事件
    ws.on('connection', (socket) => {
        // 在这里可以处理收到的消息，执行相应的逻辑
        socket.on('message', (message) => {
            // 收到消息后 给客户端发送消息
            socket.send(` ${message}:张三`);
        });
        //处理 WebSocket 连接关闭的事件
        socket.on('close', () => {
            console.log('Client disconnected');
        });
    });

    //处理 WebSocket 服务器关闭的事件
    ws.on('close', () => {
        console.log('WebSocket server is closed');
    });

    //处理 WebSocket 服务器发生错误的事件
    ws.on('error', (error) => {
        console.error('WebSocket server encountered an error:', error.message);
    });
}
module.exports = { startWebSocket };